
from openpyxl import Workbook
import os

# 确保目录存在
if not os.path.exists('static/templates'):
    os.makedirs('static/templates')

# 创建工作簿
wb = Workbook()

# 创建导入模板 sheet
ws1 = wb.active
ws1.title = "导入模板"

# 设置表头
headers = ['姓名', '性别', '出生日期', '手机号', '身份证号', '住址', '部门ID', '职位', '状态', '紧急联系人', '紧急联系电话', '备注']
ws1.append(headers)

# 添加示例数据
example_data = [
    ['张三', '1', '1990-01-01', '13800138000', '110101199001011234', '北京市朝阳区', '1', '工程师', '1', '王五', '13700137000', '备注信息'],
    ['李四', '2', '1992-02-02', '13900139000', '', '', '2', '销售', '3', '', '', '']
]
for row in example_data:
    ws1.append(row)

# 创建字段说明 sheet
ws2 = wb.create_sheet(title="字段说明")

# 添加字段说明
field_descriptions = [
    ['字段', '说明', '是否必填', '格式/示例'],
    ['姓名', '员工姓名', '是', '张三'],
    ['性别', '1-男，2-女', '是', '1'],
    ['出生日期', '出生日期', '是', '1990-01-01'],
    ['手机号', '手机号码', '是', '13800138000'],
    ['身份证号', '身份证号码', '否', '110101199001011234'],
    ['住址', '居住地址', '否', '北京市朝阳区'],
    ['部门ID', '所属部门ID', '是', '1'],
    ['职位', '职位名称', '是', '工程师'],
    ['状态', '1-在职，2-离职，3-试用期', '是', '1'],
    ['紧急联系人', '紧急联系人姓名', '否', '王五'],
    ['紧急联系电话', '紧急联系人电话', '否', '13700137000'],
    ['备注', '其他说明信息', '否', '备注信息']
]

for row in field_descriptions:
    ws2.append(row)

# 保存文件
wb.save('static/templates/employee_import_template.xlsx')